package com.ydlclass.core;

import java.util.concurrent.atomic.AtomicInteger;

/**
 * 令牌桶限流算法
 * @Author duWenJian
 * @Date 2024/3/5 16:07
 * @Version 1.0
 */
public class TokenBuketRateLimiter implements RateLimiter{
    // 首先思考令牌桶是一个容器，也可以是一个数字，
    // 按照固定速率向令牌桶中生成令牌，实现固定速率的限流

    // 代表令牌的数量，>0代表有令牌，可以进行放行，减一
    /**
     * 令牌数量
     */
    private int tokens;
    /**
     * 令牌桶的容量
     */
    private int capacity;
    /**
     * 上次放令牌的时间
     */
    private long lastTime;
    /**
     * 令牌数量
     */
    private int rate;

    public TokenBuketRateLimiter(int capacity, int rate) {
        this.tokens = capacity;
        this.capacity = capacity;
        this.rate = rate;
        lastTime = System.currentTimeMillis();
    }

    /**
     * 判断请求是否可以放行
     * @return true 放行 || false 不放行
     */
    @Override
    public synchronized boolean AllocateRequest(){
        // 1、首先给令牌桶添加令牌, 计算从现在到上一次放令牌的时间间隔
        long currentTimeMillis = System.currentTimeMillis();
        long time = currentTimeMillis - lastTime;
        System.out.println("time = " + time);
        if(time > 1000/rate){
            tokens = Math.min(capacity,(int) (tokens + time*rate/1000));
            System.out.println("tokens = " + tokens);
            this.lastTime = currentTimeMillis;
        }
        // 2、取令牌
        if(tokens > 0){
            // 使用双重检测锁
                tokens--;
                return true;

        } else {
            return false;
        }
    }

    public static void main(String[] args) {
        TokenBuketRateLimiter limiter = new TokenBuketRateLimiter(10,10);
        for (int i = 0; i <100; i++) {
            try {
                Thread.sleep(90);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println("limiter.AllocateRequest() = " + limiter.AllocateRequest());
        }
    }
}
